graphing 3D in python